package com.wft;

/**
 * 给定一个链表，删除链表的倒数第 n 个节点，并且返回链表的头结点。
 * 示例：
 * 给定一个链表: 1->2->3->4->5, 和 n = 2.
 * 当删除了倒数第二个节点后，链表变为 1->2->3->5.
 * 说明：
 * 给定的 n 保证是有效的。
 */
public class Solution19 {
    public static void main(String[] args) {
        ListNode head = new ListNode(1);
        head.next = new ListNode(2);
        head.next.next = new ListNode(3);
        printNode(removeNthFromEnd(head, 3));
    }

    public static ListNode removeNthFromEnd(ListNode head, int n) {
        ListNode p1 = head;
        ListNode p2 = head;
        ListNode p3 = null;

        for (int i = 0; i < n; i++) {
            p1 = p1.next;
        }

        while (p1 != null) {
            p1 = p1.next;
            p3 = p2;
            p2 = p2.next;
        }
        if (p3 != null) {
            p3.next = p2.next;
            return head;
        } else {
            return p2.next;
        }
    }

    public static void printNode(ListNode node) {
        System.out.print(node.val);
        while (node.next != null) {
            System.out.print("==>" + node.next.val);
            node = node.next;
        }
    }
}

class ListNode {
    int val;
    ListNode next;

    public ListNode(int x) {
        val = x;
    }
}